__name__ = "ShieldNet"
__package__ = "ShieldNet"

"""
numpy~=2.1.2
torch~=2.7.0+cu128
joblib~=1.4.2
pandas~=2.2.3
loguru~=0.7.3
scapy~=2.6.1
matplotlib~=3.10.1
scikit-learn~=1.6.1
tqdm~=4.67.1
seaborn~=0.13.2
pyarrow~=19.0.1
"""
__requires__ = [
    "numpy~=2.1.2",
    "torch~=2.7.0+cu128",
    "joblib~=1.4.2",
    "pandas~=2.2.3",
    "loguru~=0.7.3",
    "scapy~=2.6.1",
    "matplotlib~=3.10.1",
    "scikit-learn~=1.6.1",
    "tqdm~=4.67.1",
    "seaborn~=0.13.2",
    "pyarrow~=19.0.1"
]

from enum import Enum

from .Detector import *
from .ModelTrainer import *
from .PcapLoder import *


class ClassifyName(Enum):
    CLASS = 'Class'
    LABEL = 'Label'

class DDoSType(Enum):
    DrDoS_DNS = 'DrDoS_DNS'
    DrDoS_LDAP = 'DrDoS_LDAP'
    DrDoS_NTP = 'DrDoS_NTP'
    DrDoS_MSSQL = 'DrDoS_MSSQL'
    DrDoS_NetBIOS = 'DrDoS_NetBIOS'
    DrDoS_SNMP = 'DrDoS_SNMP'
    DrDoS_SSDP = 'DrDoS_SSDP'
    DrDoS_UDP = 'DrDoS_UDP'
    Syn = 'Syn'
    TFTP = 'TFTP'
    UDPLag = 'UDPLag'

BENIGN_LABEL = 'BENIGN'
stream_log_handler = StreamLogHandler()